home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_zipfile.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-10-18  |  3.4 KB  |  93 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4.  
  5. try:
  6.     import zlib
  7. except ImportError:
  8.     zlib = None
  9.  
  10. import zipfile
  11. import os
  12. import unittest
  13. from StringIO import StringIO
  14. from tempfile import TemporaryFile
  15. from test.test_support import TESTFN, run_unittest
  16. TESTFN2 = TESTFN + '2'
  17.  
  18. class TestsWithSourceFile(unittest.TestCase):
  19.     
  20.     def setUp(self):
  21.         line_gen = (lambda [outmost-iterable]: for i in [outmost-iterable]:
  22. 'Test of zipfile line %d.' % i)(range(0, 1000))
  23.         self.data = '\n'.join(line_gen)
  24.         fp = open(TESTFN, 'wb')
  25.         fp.write(self.data)
  26.         fp.close()
  27.  
  28.     
  29.     def zipTest(self, f, compression):
  30.         zipfp = zipfile.ZipFile(f, 'w', compression)
  31.         zipfp.write(TESTFN, 'another' + os.extsep + 'name')
  32.         zipfp.write(TESTFN, TESTFN)
  33.         zipfp.close()
  34.         zipfp = zipfile.ZipFile(f, 'r', compression)
  35.         self.assertEqual(zipfp.read(TESTFN), self.data)
  36.         self.assertEqual(zipfp.read('another' + os.extsep + 'name'), self.data)
  37.         zipfp.close()
  38.  
  39.     
  40.     def testStored(self):
  41.         for f in (TESTFN2, TemporaryFile(), StringIO()):
  42.             self.zipTest(f, zipfile.ZIP_STORED)
  43.         
  44.  
  45.     if zlib:
  46.         
  47.         def testDeflated(self):
  48.             for f in (TESTFN2, TemporaryFile(), StringIO()):
  49.                 self.zipTest(f, zipfile.ZIP_DEFLATED)
  50.             
  51.  
  52.     
  53.     
  54.     def tearDown(self):
  55.         os.remove(TESTFN)
  56.         os.remove(TESTFN2)
  57.  
  58.  
  59.  
  60. class OtherTests(unittest.TestCase):
  61.     
  62.     def testCloseErroneousFile(self):
  63.         fp = open(TESTFN, 'w')
  64.         fp.write('this is not a legal zip file\n')
  65.         fp.close()
  66.         
  67.         try:
  68.             zf = zipfile.ZipFile(TESTFN)
  69.         except zipfile.BadZipfile:
  70.             os.unlink(TESTFN)
  71.  
  72.  
  73.     
  74.     def testNonExistentFileRaisesIOError(self):
  75.         self.assertRaises(IOError, zipfile.ZipFile, TESTFN)
  76.  
  77.     
  78.     def testClosedZipRaisesRuntimeError(self):
  79.         data = StringIO()
  80.         zipf = zipfile.ZipFile(data, mode = 'w')
  81.         zipf.writestr('foo.txt', 'O, for a Muse of Fire!')
  82.         zipf.close()
  83.         self.assertRaises(RuntimeError, zipf.testzip)
  84.  
  85.  
  86.  
  87. def test_main():
  88.     run_unittest(TestsWithSourceFile, OtherTests)
  89.  
  90. if __name__ == '__main__':
  91.     test_main()
  92.  
  93.